"""模版加载

Keyword arguments:
package_load -- 加载模版的方法 包加载 返回一个加载好的模板
Return: return_description
"""

from jinja2 import Template, PackageLoader, Environment


class TemplateLoader:
    def __init__(self):
        pass

    def package_load(
        self, package_name: str, template_path: str, template_name: str
    ) -> Template:
        """模版加载类

        Keyword arguments:
        package_name -- 包名
        template_path -- 模版路径
        template_name -- 模版名
        Return: 返回一个加载好的模板
        """
        loader = PackageLoader(package_name=package_name, package_path=template_path)
        environment = Environment(loader=loader)
        return environment.get_template(template_name)